#pragma once

#include "CBase.h"
#include "CHero.h"

enum eUnitType{GRUNT, RANGER, SIEGE, SCOUT, AERIAL, HEAVY_AERIAL, BASE_UNIT};
class CUnit: public CBase
{
	int m_nUnitType;
	CHero* m_pHero;

public:

	CUnit();
	~CUnit();

	CHero* GetHero() {return m_pHero;}
	void SetHero(CHero* pHero);

	void Init();
	void Update(float fDelta);
	void Render();
	bool Collision(CBase* pBase);
	bool CheckDistance(CBase* pBase, float fDelta);
};